/*
 * Copyright (c) 2006-2021, RT-Thread Development Team
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Change Logs:
 * Date           Author       Notes
 * 2022-01-01     wgz       the first version
 */
#include "bsp_display.h"

#include <rtthread.h>
#include <rtdevice.h>
#include "oledfont.h"

#define IIC_SLC    55   //D8
#define IIC_SDA    56   //D9


#define OLED_SCL_Clr() rt_pin_write(IIC_SLC,PIN_LOW)//SCL
#define OLED_SCL_Set() rt_pin_write(IIC_SLC,PIN_HIGH)

#define OLED_SDA_Clr() rt_pin_write(IIC_SDA,PIN_LOW)//DIN
#define OLED_SDA_Set() rt_pin_write(IIC_SDA,PIN_HIGH)

#define OLED_CMD  0 //鍐欏懡浠�
#define OLED_DATA 1 //鍐欐暟鎹�


static uint8_t OLED_GRAM[144][8];

static void iic_gpio_init(void);
static void IIC_delay(void);
static void I2C_Start(void);
static void I2C_Stop(void);
static void I2C_WaitAck(void);
static void I2C_WaitAck(void);
static void Send_Byte(uint8_t dat);

static void OLED_WR_Byte(uint8_t dat, uint8_t mode);

static void iic_gpio_init(void)
{

    rt_pin_mode(IIC_SLC, PIN_MODE_OUTPUT_OD);
    rt_pin_mode(IIC_SDA, PIN_MODE_OUTPUT_OD);
}

static void IIC_delay(void)
{
    uint8_t temp_t = 3;
    while (temp_t--);
}

static void I2C_Start(void)
{
    OLED_SDA_Set();
    OLED_SCL_Set();
    IIC_delay();
    OLED_SDA_Clr();
    IIC_delay();
    OLED_SCL_Clr();
    IIC_delay();
}


static void I2C_Stop(void)
{
    OLED_SDA_Clr();
    OLED_SCL_Set();
    IIC_delay();
    OLED_SDA_Set();
}


static void I2C_WaitAck(void)
{
    OLED_SDA_Set();
    IIC_delay();
    OLED_SCL_Set();
    IIC_delay();
    OLED_SCL_Clr();
    IIC_delay();
}


static void Send_Byte(uint8_t dat)
{
    uint8_t i = 0;
    for (i = 0; i < 8; i++)
    {
        if(dat & 0x80)
        {
            OLED_SDA_Set();
        }
        else
        {
            OLED_SDA_Clr();
        }
        IIC_delay();
        OLED_SCL_Set();
        IIC_delay();
        OLED_SCL_Clr();
        dat <<= 1;
  }
}


static void OLED_WR_Byte(uint8_t dat, uint8_t mode)
{
    I2C_Start();
    Send_Byte(0x78);
    I2C_WaitAck();
    if(mode)
    {
        Send_Byte(0x40);
    }
    else
    {
        Send_Byte(0x00);
    }
    I2C_WaitAck();
    Send_Byte(dat);
    I2C_WaitAck();
    I2C_Stop();
}

//反显函数
void OLED_ColorTurn(uint8_t i)
{
    if(i == 0)
    {
        OLED_WR_Byte(0xA6, OLED_CMD);//锟斤拷锟斤拷锟斤拷示
    }
    if(i == 1)
    {
        OLED_WR_Byte(0xA7, OLED_CMD);//锟斤拷色锟斤拷示
    }
}

//屏幕旋转180度
void OLED_DisplayTurn(uint8_t i)
{
    if(i == 0)
        {
            OLED_WR_Byte(0xC8,OLED_CMD);
            OLED_WR_Byte(0xA1,OLED_CMD);
        }
    if(i == 1)
    {
        OLED_WR_Byte(0xC0,OLED_CMD);
        OLED_WR_Byte(0xA0,OLED_CMD);
    }
}


//开启OLED显示
void OLED_DisPlay_On(void)
{
    OLED_WR_Byte(0x8D,OLED_CMD);//电荷泵使能
    OLED_WR_Byte(0x14,OLED_CMD);//开启电荷泵
    OLED_WR_Byte(0xAF,OLED_CMD);//点亮屏幕
}

//关闭OLED显示
void OLED_DisPlay_Off(void)
{
    OLED_WR_Byte(0x8D,OLED_CMD);//电荷泵使能
    OLED_WR_Byte(0x10,OLED_CMD);//关闭电荷泵
    OLED_WR_Byte(0xAE,OLED_CMD);//关闭屏幕
}

//更新显存到OLED
void OLED_Refresh(void)
{
    uint8_t i, n;
    for(i = 0; i < 8; i++)
    {
        OLED_WR_Byte(0xb0+i,OLED_CMD); //璁剧疆琛岃捣濮嬪湴鍧�
        OLED_WR_Byte(0x00,OLED_CMD);   //璁剧疆浣庡垪璧峰鍦板潃
        OLED_WR_Byte(0x10,OLED_CMD);   //璁剧疆楂樺垪璧峰鍦板潃址
        I2C_Start();
        Send_Byte(0x78);
        I2C_WaitAck();
        Send_Byte(0x40);
        I2C_WaitAck();
        for(n = 0; n < 128; n++)
        {
            Send_Byte(OLED_GRAM[n][i]);
            I2C_WaitAck();
        }
        I2C_Stop();
  }
}
//清屏函数
void OLED_Clear(void)
{
    uint8_t i, n;
    for (i = 0; i < 8; i++)
    {
       for(n = 0; n < 128; n++)
        {
           OLED_GRAM[n][i]=0;
        }
  }
    OLED_Refresh();
}

//画点
//x:0~127
//y:0~63
//t:1 填充 0,清空
void OLED_DrawPoint(uint8_t x, uint8_t y, uint8_t t)
{
    uint8_t i,m,n;
    i = y / 8;
    m = y % 8;
    n = 1 << m;
    if( t )
    {
        OLED_GRAM[x][i]|=n;
    }
    else
    {
        OLED_GRAM[x][i] =~ OLED_GRAM[x][i];
        OLED_GRAM[x][i] |= n;
        OLED_GRAM[x][i] =~ OLED_GRAM[x][i];
    }
}

//画线
//x1,y1:起点坐标
//x2,y2:结束坐标
void OLED_DrawLine(uint8_t x1,uint8_t y1,uint8_t x2,uint8_t y2,uint8_t mode)
{
    uint16_t t;
    int xerr = 0, yerr = 0, delta_x, delta_y, distance;
    int incx,incy,uRow,uCol;
    delta_x = x2 - x1;                      //锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷
    delta_y = y2 - y1;
    uRow = x1;                              //锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟�
    uCol = y1;
    if(delta_x > 0) incx = 1;               //锟斤拷锟矫碉拷锟斤拷锟斤拷锟斤拷
    else if (delta_x == 0) incx = 0;        //锟斤拷直锟斤拷
    else {incx = -1; delta_x =- delta_x;}

    if (delta_y > 0) incy = 1;
    else if (delta_y == 0) incy = 0;        //水平锟斤拷
    else {incy = -1; delta_y = -delta_x;}

    if (delta_x > delta_y) distance = delta_x;    //选取锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷
    else distance = delta_y;

    for(t = 0; t < distance + 1; t++)
    {
        OLED_DrawPoint(uRow,uCol,mode);//锟斤拷锟斤拷
        xerr += delta_x;
        yerr += delta_y;
        if(xerr > distance)
        {
            xerr -= distance;
            uRow += incx;
        }
        if(yerr > distance)
        {
            yerr -= distance;
            uCol += incy;
        }
    }
}
////x,y:圆心坐标
////r:圆的半径
//void OLED_DrawCircle(uint8_t x, uint8_t y, uint8_t r)
//{
//    int a, b,num;
//    a = 0;
//    b = r;
//    while(2 * b * b >= r * r)
//    {
//        OLED_DrawPoint(x + a, y - b,1);
//        OLED_DrawPoint(x - a, y - b,1);
//        OLED_DrawPoint(x - a, y + b,1);
//        OLED_DrawPoint(x + a, y + b,1);
//
//        OLED_DrawPoint(x + b, y + a,1);
//        OLED_DrawPoint(x + b, y - a,1);
//        OLED_DrawPoint(x - b, y - a,1);
//        OLED_DrawPoint(x - b, y + a,1);
//
//        a++;
//        num = (a * a + b * b) - r*r;//锟斤拷锟姐画锟侥碉拷锟斤拷圆锟侥的撅拷锟斤拷
//        if(num > 0)
//        {
//            b--;
//            a--;
//        }
//    }
//}
//在指定位置显示一个字符,包括部分字符
//x:0~127
//y:0~63
//size1:选择字体 6x8/6x12/8x16/12x24
//mode:0,反色显示;1,正常显示
void OLED_ShowChar(uint8_t x, uint8_t y, uint8_t chr, uint8_t size1, uint8_t mode)
{
    uint8_t i,m,temp,size2,chr1;
    uint8_t x0=x,y0=y;
    if(size1==8)size2=6;
    else size2=(size1/8+((size1%8)?1:0))*(size1/2);  //寰楀埌瀛椾綋涓�涓瓧绗﹀搴旂偣闃甸泦鎵�鍗犵殑瀛楄妭鏁�
    chr1=chr-' ';  //璁＄畻鍋忕Щ鍚庣殑鍊贾�
    for(i=0;i<size2;i++)
    {
        if(size1==8)
              {temp=asc2_0806[chr1][i];} //璋冪敤0806瀛椾綋
        else if(size1==12)
        {temp=asc2_1206[chr1][i];} //璋冪敤1206瀛椾綋
        else if(size1==16)
        {temp=asc2_1608[chr1][i];} //璋冪敤1608瀛椾綋
        else if(size1==24)
        {temp=asc2_2412[chr1][i];} //璋冪敤2412瀛椾綋
        else return;
        for(m=0;m<8;m++)
        {
            if(temp&0x01)OLED_DrawPoint(x,y,mode);
            else OLED_DrawPoint(x,y,!mode);
            temp>>=1;
            y++;
        }
        x++;
        if((size1!=8)&&((x-x0)==size1/2))
        {x=x0;y0=y0+8;}
        y=y0;
  }
}


//显示字符串
//x,y:起点坐标
//size1:字体大小
//*chr:字符串起始地址
//mode:0,反色显示;1,正常显示
void OLED_ShowString(uint8_t x, uint8_t y, uint8_t *chr, uint8_t size1, uint8_t mode)
{
    while((*chr>=' ')&&(*chr<='~'))//判断是不是非法字符!
    {
        OLED_ShowChar(x,y,*chr,size1,mode);
        if(size1 == 8) x += 6;
        else x += size1/2;
        chr++;
  }
}

//m^n
uint32_t OLED_Pow(uint8_t m,uint8_t n)
{
    uint32_t result=1;
    while(n--)
    {
      result*=m;
    }
    return result;
}

//显示数字
//x,y :起点坐标
//num :要显示的数字
//len :数字的位数
//size:字体大小
//mode:0,反色显示;1,正常显示
void OLED_ShowNum(uint8_t x,uint8_t y,uint32_t num,uint8_t len,uint8_t size1,uint8_t mode)
{
    uint8_t t,temp,m=0;
    if(size1==8)m=2;
    for(t=0;t<len;t++)
    {
        temp=(num/OLED_Pow(10,len-t-1))%10;
            if(temp==0)
            {
                OLED_ShowChar(x+(size1/2+m)*t,y,'0',size1,mode);
      }
            else
            {

              OLED_ShowChar(x+(size1/2+m)*t,y,temp+'0',size1,mode);
            }
  }
}

//显示汉字
//x,y:起点坐标
//num:汉字对应的序号
//mode:0,反色显示;1,正常显示
//void OLED_ShowChinese(uint8_t x,uint8_t y,uint8_t num,uint8_t size1,uint8_t mode)
//{
//    uint16_t m,temp;
//    uint8_t x0=x,y0=y;
//    uint16_t i,size3=(size1/8+((size1%8)?1:0))*size1;  //锟矫碉拷锟斤拷锟斤拷一锟斤拷锟街凤拷锟斤拷应锟斤拷锟斤拷锟斤拷占锟斤拷锟街斤拷锟斤拷
//    for(i=0;i<size3;i++)
//    {
//        if(size1==16)
//                {temp = Hzk1[num][i];}//锟斤拷锟斤拷16*16锟斤拷锟斤拷
//        else if(size1==24)
//                {temp = Hzk2[num][i];}//锟斤拷锟斤拷24*24锟斤拷锟斤拷
//        else if(size1 == 32)
//                {temp = Hzk3[num][i];}//锟斤拷锟斤拷32*32锟斤拷锟斤拷
//        else if(size1 == 64)
//                {temp = Hzk4[num][i];}//锟斤拷锟斤拷64*64锟斤拷锟斤拷
//        else return;
//        for(m=0;m<8;m++)
//        {
//            if(temp&0x01)OLED_DrawPoint(x,y,mode);
//            else OLED_DrawPoint(x,y,!mode);
//            temp>>=1;
//            y++;
//        }
//        x++;
//        if((x-x0)==size1)
//        {x=x0;y0=y0+8;}
//        y=y0;
//    }
//}

////num 显示汉字的个数
////space 每一遍显示的间隔
////mode:0,反色显示;1,正常显示
//void OLED_ScrollDisplay(uint8_t num, uint8_t space, uint8_t mode)
//{
//    uint8_t i, n, t = 0, m = 0, r;
//    while(1)
//    {
//        if(m==0)
//        {
//        OLED_ShowChinese(128,24,t,16,mode); //写锟斤拷一锟斤拷锟斤拷锟街憋拷锟斤拷锟斤拷OLED_GRAM[][]锟斤拷锟斤拷锟斤拷
//            t++;
//        }
//        if(t==num)
//            {
//                for(r=0;r<16*space;r++)      //锟斤拷示锟斤拷锟�
//                 {
//                    for(i=1;i<144;i++)
//                        {
//                            for(n=0;n<8;n++)
//                            {
//                                OLED_GRAM[i-1][n]=OLED_GRAM[i][n];
//                            }
//                        }
//           OLED_Refresh();
//                 }
//        t=0;
//      }
//        m++;
//        if(m==16){m=0;}
//        for(i=1;i<144;i++)   //实锟斤拷锟斤拷锟斤拷
//        {
//            for(n=0;n<8;n++)
//            {
//                OLED_GRAM[i-1][n]=OLED_GRAM[i][n];
//            }
//        }
//        OLED_Refresh();
//    }
//}

////x,y：起点坐标
////sizex,sizey,图片长宽
////BMP[]：要写入的图片数组
////mode:0,反色显示;1,正常显示
//void OLED_ShowPicture(uint8_t x,uint8_t y,uint8_t sizex,uint8_t sizey,uint8_t BMP[],uint8_t mode)
//{
//    uint16_t j=0;
//    uint8_t i,n,temp,m;
//    uint8_t x0=x,y0=y;
//    sizey=sizey/8+((sizey%8)?1:0);
//    for(n=0;n<sizey;n++)
//    {
//         for(i=0;i<sizex;i++)
//         {
//                temp=BMP[j];
//                j++;
//                for(m=0;m<8;m++)
//                {
//                    if(temp&0x01)OLED_DrawPoint(x,y,mode);
//                    else OLED_DrawPoint(x,y,!mode);
//                    temp>>=1;
//                    y++;
//                }
//                x++;
//                if((x-x0)==sizex)
//                {
//                    x=x0;
//                    y0=y0+8;
//                }
//                y=y0;
//     }
//     }
//}

void oled_init(void)
{
    iic_gpio_init();

    rt_thread_mdelay(200);

    OLED_WR_Byte(0xAE,OLED_CMD);//--turn off oled panel
    OLED_WR_Byte(0x00,OLED_CMD);//---set low column address
    OLED_WR_Byte(0x10,OLED_CMD);//---set high column address
    OLED_WR_Byte(0x40,OLED_CMD);//--set start line address  Set Mapping RAM Display Start Line (0x00~0x3F)
    OLED_WR_Byte(0x81,OLED_CMD);//--set contrast control register
    OLED_WR_Byte(0xCF,OLED_CMD);// Set SEG Output Current Brightness
    OLED_WR_Byte(0xA1,OLED_CMD);//--Set SEG/Column Mapping     0xa0宸﹀彸鍙嶇疆 0xa1姝ｅ父
    OLED_WR_Byte(0xC8,OLED_CMD);//Set COM/Row Scan Direction   0xc0涓婁笅鍙嶇疆 0xc8姝ｅ父
    OLED_WR_Byte(0xA6,OLED_CMD);//--set normal display
    OLED_WR_Byte(0xA8,OLED_CMD);//--set multiplex ratio(1 to 64)
    OLED_WR_Byte(0x3f,OLED_CMD);//--1/64 duty
    OLED_WR_Byte(0xD3,OLED_CMD);//-set display offset   Shift Mapping RAM Counter (0x00~0x3F)
    OLED_WR_Byte(0x00,OLED_CMD);//-not offset
    OLED_WR_Byte(0xd5,OLED_CMD);//--set display clock divide ratio/oscillator frequency
    OLED_WR_Byte(0x80,OLED_CMD);//--set divide ratio, Set Clock as 100 Frames/Sec
    OLED_WR_Byte(0xD9,OLED_CMD);//--set pre-charge period
    OLED_WR_Byte(0xF1,OLED_CMD);//Set Pre-Charge as 15 Clocks & Discharge as 1 Clock
    OLED_WR_Byte(0xDA,OLED_CMD);//--set com pins hardware configuration
    OLED_WR_Byte(0x12,OLED_CMD);
    OLED_WR_Byte(0xDB,OLED_CMD);//--set vcomh
    OLED_WR_Byte(0x30,OLED_CMD);//Set VCOM Deselect Level
    OLED_WR_Byte(0x20,OLED_CMD);//-Set Page Addressing Mode (0x00/0x01/0x02)
    OLED_WR_Byte(0x02,OLED_CMD);//
    OLED_WR_Byte(0x8D,OLED_CMD);//--set Charge Pump enable/disable
    OLED_WR_Byte(0x14,OLED_CMD);//--set(0x10) disable
    OLED_Clear();
    OLED_WR_Byte(0xAF,OLED_CMD);

}
